#pragma once

#include <iostream>
#include <atomic>
#include <fstream>
#include <sys/types.h>
#include <sys/stat.h>
#include <sys/time.h>
#include <unistd.h>
#include <time.h>
#include <boost/algorithm/string.hpp>

namespace ns_util
{
    class TimeUtil
    {
    public:

        // 将时间戳转化为具体时间
         static std::string TimeStampExLocalTime()
         {
             time_t tm = time(nullptr);
             struct tm *stm = localtime(&tm);
             char buffer[128];
             snprintf(buffer, sizeof(buffer), "%d-%d-%d %02d:%02d:%02d",
                      stm->tm_year + 1900, stm->tm_mon + 1, stm->tm_mday,
                      stm->tm_hour, stm->tm_min, stm->tm_sec);
             return buffer;
         }

        // 获取时间戳
        static std::string GetTimeStamp()
        {
            struct timeval tval;
            gettimeofday(&tval, nullptr);
            return std::to_string(tval.tv_sec);
        }
        // 获取毫秒级时间戳
        static std::string GetTimeMSStamp()
        {
            struct timeval tval;
            gettimeofday(&tval, nullptr);
            return std::to_string((tval.tv_usec / 1000 + tval.tv_sec * 1000));
        }
    };

    std::string temp_path = "./temp/";
    class PathUtil
    {
    public:
        static std::string AddSuffix(const std::string &file_name, const std::string &suffix)
        {
            std::string pathname = temp_path;
            pathname += file_name;
            pathname += suffix;
            return pathname;
        }
        // 编译环节形成的临时文件
        static std::string Exe(const std::string &file_name)
        {
            return AddSuffix(file_name, ".exe");
        }
        static std::string Src(const std::string &file_name)
        {
            return AddSuffix(file_name, ".cpp");
        }
        static std::string CompileError(const std::string &file_name)
        {
            return AddSuffix(file_name, ".compile_error");
        }

        // 运行环节形成的临时文件
        static std::string Stdin(const std::string &file_name)
        {
            return AddSuffix(file_name, ".stdin");
        }
        static std::string Stdout(const std::string &file_name)
        {
            return AddSuffix(file_name, ".stdout");
        }
        static std::string Stderr(const std::string &file_name)
        {
            return AddSuffix(file_name, ".stderr");
        }
    };

    class FileUtil
    {
    public:
        static bool IsFileExist(const std::string &filename)
        {
            struct stat st;
            if (stat(filename.c_str(), &st) == 0)
            {
                return true;
            }
            return false;
        }
        // 毫秒级时间戳 + 原子性递增变量 来保证文件名的唯一性
        static std::string UniqueFileName()
        {
            static std::atomic_uint id(0);
            id++;
            std::string ms = TimeUtil::GetTimeMSStamp();
            std::string str_id = std::to_string(id);
            return ms + "_" + str_id;
        }
        // 将用户提交代码写入文件
        static bool WriteFile(const std::string &filename, const std::string &content)
        {
            std::ofstream out(filename);
            if (!out.is_open())
            {
                return false;
            }
            out.write(content.c_str(), content.size());
            out.close();
            return true;
        }
        static bool ReadFile(const std::string &filename, std::string *content, bool keep = false)
        {
            (*content).clear();
            std::ifstream in(filename);
            if (!in.is_open())
            {
                return false;
            }
            // getline 读取文本的到时候不会读取分隔符\n。所以在查看结果的时候会非常凌乱
            std::string line;
            while (std::getline(in, line))
            {
                (*content) += line;
                (*content) += keep ? "\n" : ""; // 通过选项，保留格式方便阅读
            }
            in.close();
            return true;
        }
    };

    class StringUtil
    {
    public:
        /*****************
         * 第一个参数：输入要分割的字符串
         * 第二个参数：输出型参数，输出分割后的结果
         * 第三个特定的分隔符：默认" "为分隔符
         * 返回值：是否出错
         */
        static void SplitString(std::string &in, std::vector<std::string> *out, std::string Sep = " ")
        {
            // 最后一个参数表示是否压缩
            // 1:::::3:4:5 --> 1 3 4 5
            boost::split((*out), in, boost::is_any_of(Sep), boost::algorithm::token_compress_on);
        }
    };

}